package org.jruby.ext.ffi.jffi;
import com.kenai.jffi.Function;
import com.kenai.jffi.HeapInvocationBuffer;
import org.jruby.RubyModule;
import org.jruby.runtime.ThreadContext;
import org.jruby.runtime.builtin.IRubyObject;
final class DefaultMethodTwoArg extends DefaultMethod {
private final ParameterMarshaller m1, m2;
public DefaultMethodTwoArg(RubyModule implementationClass, Function function,
FunctionInvoker functionInvoker, ParameterMarshaller[] marshallers) {
super(implementationClass, function, functionInvoker, marshallers);
m1 = marshallers[0];
m2 = marshallers[1];
}
@Override
public final IRubyObject call(ThreadContext context, IRubyObject self, RubyModule klazz, String name,
IRubyObject arg1, IRubyObject arg2) {
HeapInvocationBuffer buffer = new HeapInvocationBuffer(function);
if (needsInvocationSession) {
Invocation invocation = new Invocation(context, postInvokeCount, referenceCount);
try {
m1.marshal(invocation, buffer, arg1);
m2.marshal(invocation, buffer, arg2);
return functionInvoker.invoke(context, function, buffer);
} finally {
invocation.finish();
}
} else {
m1.marshal(context, buffer, arg1);
m2.marshal(context, buffer, arg2);
return functionInvoker.invoke(context, function, buffer);
}
}
}